import torch
# 数据集路径
datapath = './data/eng-fra-v2.txt'

# 指定特殊token
SOS_token = 0  # 开始字符
EOS_token = 1  # 结束字符

# 指定句子的最大长度，包含标点（一般需要提前对语料进行长度分析）
MAX_LENGTH = 10

# 指定训练设备
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")

# 训练策略分数
teacher_forcing_ratio = 0.5

# 学习率
my_lr = 1e-4

# 训练轮次
epochs = 2